package club.xiaojiawei.greedy;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 5/26/22 2:28 PM
 * @question 122. 买卖股票的最佳时机 II
 * @description 给你一个整数数组 prices ，其中 prices[i] 表示某支股票第 i 天的价格。
 * 在每一天，你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买，然后在 同一天 出售。
 * 返回 你能获得的 最大 利润 。
 */
public class MaxProfit122 {

    public static void main(String[] args) {
        MaxProfit122 test = new MaxProfit122();
        int result = test.maxProfit(new int[]{7, 1, 5, 3, 2, 3});
        System.out.println("最大利润："+result);
    }

    /**
     * 贪心
     * @param prices
     * @return
     */
    public int maxProfit(int[] prices) {
        boolean flag = true;
        int temp = Integer.MAX_VALUE;
        int count = 0;
        for (int i = 0; i < prices.length; i++) {
            if (flag){
                if (prices[i] > temp){
                    count -= prices[i - 1];
                    flag = false;
                }
            }else {
                if (prices[i] < temp){
                    count += prices[i - 1];
                    flag = true;
                }
            }
            temp = prices[i];
        }
        if (!flag){
            count += prices[prices.length - 1];
        }
        return count;
    }

    /**
     * 官方-dp
     * 转移方程： dp[i][0]=max{dp[i−1][0],dp[i−1][1]+prices[i]}   dp[i][0] 表示第 ii 天交易完后手里没有股票的最大利润
     *           dp[i][1]=max{dp[i−1][1],dp[i−1][0]−prices[i]}  dp[i][1] 表示第 ii 天交易完后手里持有一支股票的最大利润（ii 从 00 开始）
     * @param prices
     * @return
     */
    public int maxProfit2(int[] prices) {
        int n = prices.length;
        int[][] dp = new int[n][2];
        dp[0][0] = 0;
        dp[0][1] = -prices[0];
        for (int i = 1; i < n; ++i) {
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
        }
        return dp[n - 1][0];
    }

    /**
     * 官方-dp
     * @param prices
     * @return
     */
    public int maxProfit3(int[] prices) {
        int n = prices.length;
        int dp0 = 0, dp1 = -prices[0];
        for (int i = 1; i < n; ++i) {
            int newDp0 = Math.max(dp0, dp1 + prices[i]);
            int newDp1 = Math.max(dp1, dp0 - prices[i]);
            dp0 = newDp0;
            dp1 = newDp1;
        }
        return dp0;
    }

    /**
     * 官方-贪心
     * @param prices
     * @return
     */
    public int maxProfit4(int[] prices) {
        int ans = 0;
        int n = prices.length;
        for (int i = 1; i < n; ++i) {
//            只要能赚就买抛，不需要等到价格最低或最高再买抛
            ans += Math.max(0, prices[i] - prices[i - 1]);
        }
        return ans;
    }
}
